# -*- coding: utf-8 -*-
'''
Created on 2017年2月27日

@author: ZhuJiahui
'''

import jieba.analyse

def get_cn_keyword(text_corpus, keyword_num=5):
    '''
    采用Jieba分词提取中文关键词
    :param text_corpus: 文档集合 (2d string list)  不能带有词性标注
    :param keyword_num: 关键词数量
    :return 所有文档的所有关键词列表 (2d string list)
             所有文档中出现的每一个不同的关键词构成的列表 即全局关键词 (1d string list)
    '''
    
    global_keyword_list = []
    all_keyword_list = []
    for i in range(len(text_corpus)):
        this_keywords = jieba.analyse.extract_tags(" ".join(text_corpus[i]), keyword_num)
        all_keyword_list.append(this_keywords)
        for this_word in this_keywords:
            if this_word not in global_keyword_list:
                global_keyword_list.append(this_word)
    
    return all_keyword_list, global_keyword_list


def get_cn_global_keyword(text_corpus, keyword_num=5):
    '''
    采用Jieba分词提取中文关键词
    :param text_corpus: 文档集合 (2d string list)  不能带有词性标注
    :param keyword_num: 关键词数量
    :return 所有文档的所有关键词列表 (2d string list)
             所有文档中出现的每一个不同的关键词构成的列表 即全局关键词 (1d string list)
    '''
    
    global_keyword_list = []
    for i in range(len(text_corpus)):
        this_keywords = jieba.analyse.extract_tags(" ".join(text_corpus[i]), keyword_num)
        for this_word in this_keywords:
            if this_word not in global_keyword_list:
                global_keyword_list.append(this_word)
    
    return global_keyword_list

if __name__ == '__main__':
    print(get_cn_global_keyword([["李克强", "是", "温家宝", "的", "继任", "是", "总理"]], 3))